from selenium.webdriver.common.by import By

from utils.page_base import PageBase


#UI：登录页
class PageLogin(PageBase):
    url="/simple/login"
    loc_zanhao=By.NAME,"login_info"
    loc_pwd=By.NAME,"password"
    loc_login=By.CSS_SELECTOR,"input[value='登录']"
    loc_errorTip=By.CSS_SELECTOR,"div.prompt"

    def __init__(self, driver, base_url):
        PageBase.__init__(self, driver, base_url)

    #函数：打开当前页面
    def openUrl(self):
        self.open(PageLogin.url)

    #函数：输入账号
    def inputZanhao(self,data):
        self.input(PageLogin.loc_zanhao,data)


    #函数：输入密码
    def inputPwd(self,data):
        self.input(PageLogin.loc_pwd,data)

    #函数：点击登录按钮
    def click_login(self):
        self.click(PageLogin.loc_login)


    #函数：获取错误提示
    def getErrorTips(self):
        return self.getText(PageLogin.loc_errorTip)


    #函数：获取密码框的class属性的值
    def getPwdClassAttr(self):
        return self.getAttr(PageLogin.loc_pwd,"class")

    # 函数：获取账号框的class属性的值
    def getZanhaoClassAttr(self):
        return self.getAttr(PageLogin.loc_zanhao, "class")